/* CoralCube | Massively Multiplayer Online Role-Playing Game(MMORPG) Emulator.
|* 
|* Copyright (C) 2005-2010 MaNGOS <http://getmangos.com/>
|*               2008-2010 TrinityCore <http://www.trinitycore.org/>
|*               2008-2010 CoralCube <http://www.CoralCube.com/>
|*
|* This program is free software. You can redistribute it and/or modify it under the terms of 
|* the GNU General Public License as published by the Free Software Foundation, either 
|* version 2 of the License.
|*
|* This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY, 
|* without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. 
|* See the GNU General Public License for more details.
|*
|* You should have received a copy of the GNU General Public License along with this program. 
|* If not, see <http://www.gnu.org/licenses/>.
|*/

#ifndef CORE_PETAI_H
#define CORE_PETAI_H

#include "CreatureAI.h"
#include "Timer.h"

class Creature;
class Spell;

class PetAI : public CreatureAI
{
public:

    explicit PetAI(Creature *c);

    void EnterEvadeMode();
    void JustDied(Unit * /*who*/) { _stopAttack(); }

    void UpdateAI(const uint32);
    static int Permissible(const Creature *);

    void KilledUnit(Unit * /*victim*/);
    void AttackStart(Unit *target);
    void MovementInform(uint32 moveType, uint32 data);

private:
    bool _isVisible(Unit *) const;
    bool _needToStop(void);
    void _stopAttack(void);

    void UpdateAllies();

    TimeTracker i_tracker;
    bool inCombat;
    bool targetHasCC;
    std::set<uint64> m_AllySet;
    uint32 m_updateAlliesTimer;

    Unit *SelectNextTarget();
    void HandleReturnMovement();
    void DoAttack(Unit *target, bool chase);
    bool _CanAttack(Unit *target);
    bool _CheckTargetCC(Unit *target);
};
#endif
